package machine_learning.weka;

import weka.classifiers.trees.J48;
import weka.core.Instances;
import weka.core.converters.ConverterUtils.DataSource;

public class WekaExample {
    public static void main(String[] args) throws Exception {
        // 1. 加载数据
        DataSource source = new DataSource("D:\\机器学习\\weka-3-8-6\\data\\iris.arff");
        Instances data = source.getDataSet();
        data.setClassIndex(data.numAttributes() - 1); // 设置类别属性

        // 2. 初始化分类器（决策树）
        J48 tree = new J48();
        tree.buildClassifier(data); // 训练模型

        // 3. 输出模型
        System.out.println(tree);
    }
}

